ubuntu查看安装的pytorch/cuda版本
使用命令:
user@home:~$ python Python 3.6.3 |Anaconda custom (64-bit)| (default, Oct 13 2017, 12:02:49) [GCC 7.2.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> torch.__version__ '1.0.1.post2'
和
user@home:~$ cat /usr/local/cuda/version.txt CUDA Version 9.0.176